
gpu=$1

if (($1==0));then
./train.py --img-list ./data/MMHWS/ct_train_resize_image_list.txt --model-dir ./models/MMHWS/image --gpu $gpu --batch-size 16 --epochs 120

elif (($1==1));then
./train.py --img-list ./data/MMHWS/ct_train_resize_label_list.txt --model-dir ./models/MMHWS/label --gpu $gpu --batch-size 16 --epochs 120
elif (($1==2));then
./register.py --moving ./data/MMHWS/ct_test1_resize/ct_test_2001_image.nii.gz --fixed ./data/MMHWS/ct_train1_resize/ct_train_1001_image.nii.gz --moved ./data/MMHWS/ct_resize_moved/ct_test_2001_image.nii.gz --model ./models/MMHWS/image/0120.pt --gpu 0
elif (($1==3));then
./train.py --img-list ./data/MMHWS/ct_train_resize_label_list.txt --model-dir ./models/MMHWS/label --gpu $gpu --batch-size 16 --epochs 120

fi